import random,requests,time,re,csv



#返回一个随机的请求头 headers
def get_UserAgent():
        USER_AGENTS = ["Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; AcooBrowser; .NET CLR 1.1.4322; .NET CLR 2.0.50727)",
                "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0; Acoo Browser; SLCC1; .NET CLR 2.0.50727; Media Center PC 5.0; .NET CLR 3.0.04506)",
                "Mozilla/4.0 (compatible; MSIE 7.0; AOL 9.5; AOLBuild 4337.35; Windows NT 5.1; .NET CLR 1.1.4322; .NET CLR 2.0.50727)",
                "Mozilla/5.0 (Windows; U; MSIE 9.0; Windows NT 9.0; en-US)",
                "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Win64; x64; Trident/5.0; .NET CLR 3.5.30729; .NET CLR 3.0.30729; .NET CLR 2.0.50727; Media Center PC 6.0)",
                "Mozilla/5.0 (compatible; MSIE 8.0; Windows NT 6.0; Trident/4.0; WOW64; Trident/4.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; .NET CLR 1.0.3705; .NET CLR 1.1.4322)",
                "Mozilla/4.0 (compatible; MSIE 7.0b; Windows NT 5.2; .NET CLR 1.1.4322; .NET CLR 2.0.50727; InfoPath.2; .NET CLR 3.0.04506.30)",
                "Mozilla/5.0 (Windows; U; Windows NT 5.1; zh-CN) AppleWebKit/523.15 (KHTML, like Gecko, Safari/419.3) Arora/0.3 (Change: 287 c9dfb30)",
                "Mozilla/5.0 (X11; U; Linux; en-US) AppleWebKit/527+ (KHTML, like Gecko, Safari/419.3) Arora/0.6",
                "Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.8.1.2pre) Gecko/20070215 K-Ninja/2.1.1",
                "Mozilla/5.0 (Windows; U; Windows NT 5.1; zh-CN; rv:1.9) Gecko/20080705 Firefox/3.0 Kapiko/3.0",
                "Mozilla/5.0 (X11; Linux i686; U;) Gecko/20070322 Kazehakase/0.4.5",
                "Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.0.8) Gecko Fedora/1.9.0.8-1.fc10 Kazehakase/0.5.6",
                "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11",
                "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_3) AppleWebKit/535.20 (KHTML, like Gecko) Chrome/19.0.1036.7 Safari/535.20",
                "Opera/9.80 (Macintosh; Intel Mac OS X 10.6.8; U; fr) Presto/2.9.168 Version/11.52",]
        UserAgent = random.choice(USER_AGENTS)
        headers = {'User-Agent': UserAgent}
        return headers

def get_random_header():
    headers= get_UserAgent()
    return headers

def scraw_proxies(page_num,scraw_url="http://www.xicidaili.com/nt/"):
    scraw_ip=list()
    available_ip=list()
    for page in range(1,page_num):
        print("抓取第%d页代理IP" %page)
        url=scraw_url+str(page)
        r=requests.get(url,headers=get_random_header())
        r.encoding='utf-8'
        pattern = re.compile('<td class="country">.*?alt="Cn" />.*?</td>.*?<td>(.*?)</td>.*?<td>(.*?)</td>', re.S)
        scraw_ip= re.findall(pattern, r.text)
        for ip in scraw_ip:
            if(test_ip(ip)==True):
                print('%s:%s通过测试，添加进可用代理列表' %(ip[0],ip[1]))
                available_ip.append(ip)
            else:
                pass    
        print("代理爬虫暂停10s")
        time.sleep(5)
        print("爬虫重启")
    print('抓取结束')
    return available_ip

def test_ip(ip,test_url='http://2017.ip138.com/ic.asp',time_out = 5):
    proxies={'http': ip[0]+':'+ip[1]}
    try_ip=ip[0]
    csvFile3 = open('Good_IP.csv','a', newline='') 
    writer2 = csv.writer(csvFile3)
    #print(try_ip)
    try:
       
        r=requests.get(test_url,headers=get_random_header(),proxies=proxies,timeout=time_out)
        if r.status_code==200:
            r.encoding='gbk'
            result=re.search('\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}',r.text)
            result=result.group()
           
            if result[:9]==try_ip[:9]:
                res = ('测试通过',)
                res1 = ip + res
                writer2.writerow(res1)
                writer2.close()
                print(r.text)
                print('测试通过')
                return True
            else:
                res = ('%s:%s 携带代理失败,使用了本地IP' %(ip[0],ip[1]),)
                res1 = ip + res
                writer2.writerow(res1)
                writer2.close()
                print('%s:%s 携带代理失败,使用了本地IP' %(ip[0],ip[1]))
                return False   
        else:
            res = ('%s:%s 请求码不是200' %(ip[0],ip[1]),)
            res1 = ip + res
            writer2.writerow(res1)
            writer2.close()
            print('%s:%s 请求码不是200' %(ip[0],ip[1]))
            return False
    except:
        res = ('%s:%s 请求过程错误' %(ip[0],ip[1]),)
        res1 = ip + res
        writer2.writerow(res1)
        #writer2.close()
        print('%s:%s 请求过程错误' %(ip[0],ip[1]))
        return False
    
    

if __name__=="__main__":
    available_ip=scraw_proxies(10)
